package com.smh.dao;

import com.smh.pojo.ClassInfo;
import org.springframework.stereotype.Repository;

import java.util.ArrayList;
import java.util.List;

@Repository
public class ClassDao {
    private List<ClassInfo> classInfoList = new ArrayList<>();

    public String add(ClassInfo classInfo) {
        for (int i = 0; i < classInfoList.size(); i++) {
            if (classInfo.getId().equals(classInfoList.get(i).getId())){
                return "error";
            }
        }
        classInfoList.add(classInfo);
        return "ok";
    }

    public List<ClassInfo> search() {
        return classInfoList;
    }

    public void delete(ClassInfo classInfo) {
        for (int i = 0; i < classInfoList.size(); i++) {
            if (classInfo.getId().equals(classInfoList.get(i).getId())){
                classInfoList.remove(i);
            }
        }
    }

    public void update(ClassInfo classInfo) {
        for (int i = 0; i < classInfoList.size(); i++) {
            ClassInfo classInfo1 = classInfoList.get(i);
            if (classInfo.getId().equals(classInfo1.getId())){
                classInfo1.setClassName(classInfo.getClassName());
                classInfo1.setDate(classInfo.getDate());
                classInfo1.setMajor(classInfo.getMajor());
            }
        }
    }

    public ClassInfo query(ClassInfo classInfo) {
        for (int i = 0; i < classInfoList.size(); i++) {
            ClassInfo classInfo1 = classInfoList.get(i);
            if (classInfo1.getId().equals(classInfo.getId())){
                return classInfo1;
            }
        }
        return null;
    }
}
